#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Date    : 2025-02-01-22:28
# @Author  :
# @File    : predict.py
# @Software: PyCharm

# 导入ultralytics的YOLO库
from ultralytics import YOLO

"""
there
预测模型（Prediction）
"""

# 加载模型
model = YOLO('ultralytics/gui/main/best.pt')  # 加载自定义的训练模型

if __name__ == '__main__':

    results = model.predict(
        r"交通视频.mp4",
        save=True,
        conf=0.5,
        device=0,
        project='runs/predict', # 输出路径
        iou=0.5)

